#include "icmp.h"
#include "ip.h"

#define PING_TIMEOUT 1000 // ms

int ping(const char* host, int cnt)
{
    //    static uint16_t seq = 0;
    //	uint16_t pid16 = (uint16_t)getpid();
    //    char ip[64] = {0};
    //    uint32_t start_tick, end_tick;
    //    uint64_t start_hrtime, end_hrtime;
    //    int timeout = 0;
    //    int sendbytes = 64;
    //    char sendbuf[64];
    //    char recvbuf[128]; // iphdr + icmp = 84 at least
    //    icmp_t* icmp_req = (icmp_t*)sendbuf;
    //    iphdr_t* ipheader = (iphdr_t*)recvbuf;
    //    icmp_t* icmp_res;
    //    // ping stat
    //    int send_cnt = 0;
    //    int recv_cnt = 0;
    //    int ok_cnt = 0;
    //    float rtt, min_rtt, max_rtt, total_rtt;
    //    rtt = max_rtt = total_rtt = 0.0f;
    //    min_rtt = 1000000.0f;
    //    //min_rtt = MIN(rtt, min_rtt);
    //    //max_rtt = MAX(rtt, max_rtt);
    //    // gethostbyname -> socket -> setsockopt -> sendto -> recvfrom -> closesocket
    //    sockaddr_u peeraddr;
    //    socklen_t addrlen = sizeof(peeraddr);
    //    memset(&peeraddr, 0, addrlen);
    //    int ret = Resolver(host, &peeraddr);
    //    if (ret != 0) return ret;
    //    sockaddr_ip(&peeraddr, ip, sizeof(ip));
    //    int sockfd = socket(peeraddr.sa.sa_family, SOCK_RAW, IPPROTO_ICMP);
    //    if (sockfd < 0) {
    //        perror("socket");
    //        if (errno == EPERM) {
    //            fprintf(stderr, "please use root or sudo to create a raw socket.\n");
    //        }
    //        return -socket_errno();
    //    }
    //
    //    timeout = PING_TIMEOUT;
    //    ret = so_sndtimeo(sockfd, timeout);
    //    if (ret < 0) {
    //        perror("setsockopt");
    //        goto error;
    //    }
    //    timeout = PING_TIMEOUT;
    //    ret = so_rcvtimeo(sockfd, timeout);
    //    if (ret < 0) {
    //        perror("setsockopt");
    //        goto error;
    //    }
    //
    //    icmp_req->icmp_type = ICMP_ECHO;
    //    icmp_req->icmp_code = 0;
    //    icmp_req->icmp_id = pid16;
    //    for (int i = 0; i < sendbytes - sizeof(icmphdr_t); ++i) {
    //        icmp_req->icmp_data[i] = i;
    //    }
    //    start_tick = gettick_ms();
    //    while (cnt-- > 0) {
    //        // NOTE: checksum
    //        icmp_req->icmp_seq = ++seq;
    //        icmp_req->icmp_cksum = 0;
    //        icmp_req->icmp_cksum = checksum((uint8_t*)icmp_req, sendbytes);
    //        start_hrtime = gethrtime_us();
    //        addrlen = sockaddr_len(&peeraddr);
    //        int nsend = sendto(sockfd, sendbuf, sendbytes, 0, &peeraddr.sa, addrlen);
    //        if (nsend < 0) {
    //            perror("sendto");
    //            continue;
    //        }
    //        ++send_cnt;
    //        addrlen = sizeof(peeraddr);
    //        int nrecv = recvfrom(sockfd, recvbuf, sizeof(recvbuf), 0, &peeraddr.sa, &addrlen);
    //        if (nrecv < 0) {
    //            perror("recvfrom");
    //            continue;
    //        }
    //        ++recv_cnt;
    //        end_hrtime = gethrtime_us();
    //        // check valid
    //        bool valid = false;
    //        int iphdr_len = ipheader->ihl * 4;
    //        int icmp_len = nrecv - iphdr_len;
    //        if (icmp_len == sendbytes) {
    //            icmp_res = (icmp_t*)(recvbuf + ipheader->ihl*4);
    //            if (icmp_res->icmp_type == ICMP_ECHOREPLY &&
    //                icmp_res->icmp_id == pid16 &&
    //                icmp_res->icmp_seq == seq) {
    //                valid = true;
    //            }
    //        }
    //        if (valid == false) {
    //            printd("recv invalid icmp packet!\n");
    //            continue;
    //        }
    //        rtt = (end_hrtime-start_hrtime) / 1000.0f;
    //        min_rtt = MIN(rtt, min_rtt);
    //        max_rtt = MAX(rtt, max_rtt);
    //        total_rtt += rtt;
    //        printd("%d bytes from %s: icmp_seq=%u ttl=%u time=%.1f ms\n", icmp_len, ip, seq,
    //        ipheader->ttl, rtt); fflush(stdout);
    //        ++ok_cnt;
    //        if (cnt > 0) hv_sleep(1); // sleep a while, then agian
    //    }
    //    end_tick = gettick_ms();
    //    printd("--- %s ping statistics ---\n", host);
    //    printd("%d packets transmitted, %d received, %d%% packet loss, time %d ms\n",
    //        send_cnt, recv_cnt, (send_cnt-recv_cnt)*100/(send_cnt==0?1:send_cnt),
    //        end_tick-start_tick);
    //    printd("rtt min/avg/max = %.3f/%.3f/%.3f ms\n",
    //        min_rtt, total_rtt/(ok_cnt==0?1:ok_cnt), max_rtt);
    //
    //    closesocket(sockfd);
    //    return ok_cnt;
    // error:
    //    closesocket(sockfd);
    //    return socket_errno() > 0 ? -socket_errno() : -1;
    return -1;
}
